/*
 * Copyright 2017 StreamSets Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.streamsets.pipeline.stage.kinetica.util;

import java.util.HashMap;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.gpudb.GPUdb;
import com.gpudb.GPUdbBase;
import com.gpudb.GPUdbException;
import com.gpudb.protocol.ShowSystemStatusResponse;
import com.streamsets.pipeline.api.StageException;
import com.streamsets.pipeline.stage.kinetica.KineticaConfigBean;

public class KineticaConnectionUtils {

  private final static Logger LOG = LoggerFactory.getLogger(KineticaConnectionUtils.class);

  public GPUdb getGPUdb(KineticaConfigBean conf) throws GPUdbException, StageException {

    GPUdbBase.Options gpudbOptions = new GPUdbBase.Options();

    if (conf.transportCompressionEnabled) {
      gpudbOptions.setUseSnappy(true);
    } else {
      gpudbOptions.setUseSnappy(false);
    }

    if (conf.username != null && conf.password != null) {
      gpudbOptions.setUsername(conf.username.get());
      gpudbOptions.setPassword(conf.password.get());
    }

    GPUdb gpudb = new GPUdb(conf.url, gpudbOptions);
    ShowSystemStatusResponse sssr = gpudb.showSystemStatus(new HashMap<String, String>());

    LOG.info("Attempting connection to Kinetica at " + conf.url + "...");
    if (sssr == null || sssr.getStatusMap() == null || sssr.getStatusMap().size() == 0) {
      throw new GPUdbException("Error connecting to Kinetica");
    }

    LOG.info("Connection is good!");

    if (conf.transportCompressionEnabled) {
      LOG.info("Transport Compression is enabled");
    } else {
      LOG.info("Transport Compression is not enabled");
    }

    return gpudb;
  }
}
